package br.com.virtual.base.uteis;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

import br.com.virtual.cadastrar.model.Produto;

public class ProdutoDataModel extends ListDataModel<Produto> implements
		SelectableDataModel<Produto> {

	public ProdutoDataModel() {
		super();
	}

	public ProdutoDataModel(List<Produto> list) {
		super(list);
	}

	@Override
	public Produto getRowData(String arg0) {

		@SuppressWarnings("unchecked")
		List<Produto> produtos = (List<Produto>) getWrappedData();

		for (Produto end : produtos) {
			if (end.getId().toString().equals(arg0))
				return end;
		}

		return null;
	}

	@Override
	public Object getRowKey(Produto produto) {
		return produto.getId();
	}

}
